package main

import (
	"fmt"
	"gorm.io/gorm/logger"
	"gorm.io/gorm/schema"
	"log"
	"os"
	"time"

	_ "github.com/go-sql-driver/mysql"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

var db *gorm.DB

//sqlx 是对 sql 进行的一层封装
func initializeDatabase() (err error) {
	newLogger := logger.New(
		log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer（日志输出的目标，前缀和日志包含的内容——译者注）
		logger.Config{
			SlowThreshold:             time.Second, // 慢 SQL 阈值
			LogLevel:                  logger.Info, // 日志级别
			IgnoreRecordNotFoundError: true,        // 忽略ErrRecordNotFound（记录未找到）错误
			Colorful:                  true,        // 禁用彩色打印
		},
	)
	//data source name
	//用户名@密码@协议(ip:port)/数据库
	dsn := "root:123456@tcp(gincloud:3306)/gin_demo?charset=utf8mb4&parseTime=True&loc=Local"
	db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
		//日志配置
		Logger: newLogger,
		//自定义表名格式
		NamingStrategy: schema.NamingStrategy{
			//自定义表名前缀
			TablePrefix: "t_",
			//单数模式(表名后不加s)
			SingularTable: true,
		},
	})
	if err != nil {
		panic(err)
	}
	return nil
}

type Product struct {
	//gorm.Model 封装了id(主键), 创建/更新时间
	gorm.Model
	Code  string
	Price uint
}

//自定义表名
/*func (Product)TableName() string {
	//优先级最高, 会覆盖通用配置(表名前缀/后缀...)
	return "t_udf_product"
}
*/

func main() {
	//连接数据库
	if err := initializeDatabase(); err != nil {
		panic(err)
	}
	fmt.Println("connected")

	// 迁移 schema(根据 Product 结构自动创建表)
	db.AutoMigrate(&Product{})

	// Create
	db.Create(&Product{Code: "D42", Price: 100})

	// Read
	var product Product
	// 根据整形主键查找
	db.First(&product, 1)
	// 查找 code 字段值为 D42 的记录
	db.First(&product, "code = ?", "D42")

	// Query - 将 product 的 price 更新为 200
	db.Model(&product).Update("Price", 200)
	// Query - 更新多个字段
	// 仅更新非零值字段,零值需特殊定义 struct 中字段类型
	db.Model(&product).Updates(Product{Price: 200, Code: "F42"})
	//使用 map 方式, 不使用结构体
	db.Model(&product).Updates(map[string]interface{}{"Price": 200, "Code": "F42"})

	// Delete - 删除 product (伪删除, 软删除)
	// 因为结构体引用了 gorm.Model, gorm.Model中包含了 DeletedAt 类型字段, 将自动解析软删除功能
	db.Delete(&product, 1)
}
